package com.scitc.utils;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.concurrent.TimeUnit;

/**
 * @author: zk
 * @date: 2021/6/20 14:39
 * @description: 根据获取请求中的ip地址
 */
@Component
public class IpUtil {

    @Autowired
    private IpUtil ipUtil;

    public String getIpAddr(HttpServletRequest request) {
        String ipAddress = null;
        try {
            ipAddress = request.getHeader("x-forwarded-for");
            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
                ipAddress = request.getHeader("Proxy-Client-IP");
            }
            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
                ipAddress = request.getHeader("WL-Proxy-Client-IP");
            }
            if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
                ipAddress = request.getRemoteAddr();
                if (ipAddress.equals("127.0.0.1")) {
                    // 根据网卡取本机配置的IP
                    InetAddress inet = null;
                    try {
                        inet = InetAddress.getLocalHost();
                    } catch (UnknownHostException e) {
                        e.printStackTrace();
                    }
                    ipAddress = inet.getHostAddress();
                }
            }
            // 对于通过多个代理的情况，第一个IP为客户端真实IP,多个IP按照','分割
            if (ipAddress != null && ipAddress.length() > 15) { // "***.***.***.***".length()
                // = 15
                if (ipAddress.indexOf(",") > 0) {
                    ipAddress = ipAddress.substring(0, ipAddress.indexOf(","));
                }
            }
        } catch (Exception e) {
            ipAddress="";
        }

        return ipAddress;
    }

    /**
     * 处理 ip
     * @param request
     * @return 1：有次数并次数减 1；0：次数不足
     */
    public int unLoginIpCheck(HttpServletRequest request, RedisTemplate<String, Integer> redisTemplate) {
        String ip = ipUtil.getIpAddr(request);
        if(redisTemplate.hasKey(ip) == false) {
            redisTemplate.opsForValue().set(ip, 4, 1, TimeUnit.DAYS);
        }else{
            int time = redisTemplate.opsForValue().get(ip);
            if(time <= 0) {
                return 0;
            }else{
                // 获取过期时间
                Long restTime = redisTemplate.opsForValue().getOperations().getExpire(ip);
                redisTemplate.opsForValue().set(ip, --time, restTime, TimeUnit.SECONDS);
            }
        }

        return 1;
    }

}
